import string
import urllib.request
from urllib.parse import quote
import pymysql
from bs4 import BeautifulSoup
def nospace(world):
    return world.strip().replace(" ", "").replace("\n", "")
city = input("请输入城市：")
code = {
'北京': '530',
    '上海': '538',
    '广州': '763',
    '深圳': '765',
    '天津': '531',
    '武汉': '736',
    '西安': '854',
    '成都': '801',
    '沈阳': '599',
    '南京': '635',
    '杭州': '653',
    '苏州': '639',
    '重庆': '551',
    '长沙': '749',
    '厦门': '682',
    '南昌': '691'
}
city = code[city]
type = input("请输入岗位：")
url = "https://sou.zhaopin.com/?jl=" + city + "&kw=" + type + "&p=1"
url = quote(url,safe=string.printable)
res = urllib.request.urlopen(url)
print(url)
content = res.read().decode()
soup = BeautifulSoup(content,"lxml")
jobitem = soup.select(".joblist-box__item")
print(jobitem)
job_list = []
for job in jobitem:
    job_dic = {}
    #get_text()
    #getText()
    job_name = job.select(".jobinfo__name")[0].get_text()
    job_sal = job.select(".jobinfo__salary")[0].get_text()
    # job_sal = job_sal.strip().replace(" ","").replace("\n","")
    job_sal = nospace(job_sal)
    print(job_sal)
    if '天' in job_sal:
        job_sal = int(job_sal.split("-")[0])*30
    elif('千' in job_sal):
        job_sal = int(job_sal.split("千")[0])*1000
    elif('万' in job_sal):
        job_sal = float(job_sal.split("万")[0]) * 10000
    else:
        job_sal = 5000
    company_name = nospace(job.select(".companyinfo__name")[0].get_text())
    job_msg = job.select(".jobinfo__other-info-item")
    job_place = nospace(job_msg[0].get_text())[0:2]
    job_exp = nospace(job_msg[1].get_text())
    job_education = nospace(job_msg[2].get_text())
    job_dic["job_name"] = job_name
    job_dic["job_sal"] = job_sal
    job_dic["company_name"] = company_name
    job_dic["job_place"] = job_place
    job_dic["job_exp"] = job_exp
    job_dic["job_education"] = job_education
    print(job_dic)
    job_list.append(job_dic)
db = pymysql.connect(
    host="localhost",
    port=3306,
    user="root",
    password="komirou",
    database="crawlerdemo",
    charset="utf8"
)
cursor = db.cursor()
for job in job_list:
    sql = "insert into job values(null,'%s','%s','%s','%s','%s','%s')" % (job["job_name"],job["job_sal"],job["company_name"],
                                                                          job["job_place"],job["job_exp"],job["job_education"])
    try:
        cursor.execute(sql)
    except Exception as e:
        print(e)
        print("插入数据失败")
        db.rollback()
db.commit()
cursor.close()
db.close()


